import torch
print(torch.__version__)
print("Is CUDA available: ", torch.backends.mps.is_available())